"""
@author: xiangping
@contact: xiangpingbu@gmail.com
@time: 2020/1/4 6:15 下午
@file: base_model.py
@Desc: Abstract class for AI Model
"""

from app.configs.model_path import MODEL_DICT


class BaseModel:
    model_name = "base_abstract_model"
    model_path = None
    model_instance = None

    def __init__(self, model_name):
        self.model_name = model_name
        self.model_path = MODEL_DICT.get(self.model_name)
        self.model_instance = self.load_model()

    def load_model(self):
        pass

    def process(self,**kwargs):
        pass
